package com.oj.common.core.utils;

import cn.hutool.core.util.StrUtil;
import com.oj.common.core.constants.HttpConstants;
import com.oj.common.core.constants.JwtConstants;
import io.jsonwebtoken.Claims;
import io.jsonwebtoken.Jwts;
import io.jsonwebtoken.SignatureAlgorithm;
import org.springframework.util.CollectionUtils;
import org.springframework.util.StringUtils;

import java.util.Map;

/**
 * @Description
 * @Author bin
 * @Date 2025/4/14
 */
public class JwtUtils {
    /**
     * ⽣成令牌
     *
     * @param claims 数据
     * @param secret 密钥
     * @return 令牌
     */
    public static String createToken(Map<String, Object> claims, String secret) {
        String token = Jwts.builder().setClaims(claims).signWith(SignatureAlgorithm.HS512, secret).compact();
        return token;
    }

    /**
     * 从令牌中获取数据
     *
     * @param token  令牌
     * @param secret 密钥
     * @return 数据
     */
    public static Claims parseToken(String token, String secret) {
        token = replaceTokenPrefix(token);
        //默认使用的HS512是默认的对称加密算法
        return Jwts.parser().setSigningKey(secret).parseClaimsJws(token).getBody();
    }

    public static String getUserKey(Claims claims) {
        if (claims != null) {
            String s = claims.get(JwtConstants.LOGIN_USER_KEY, String.class);
            if (s == null) {
                return "";
            }
            return s;
        }
        return null;
    }

    public static String getUserId(Claims claims) {
        if (claims != null) {
            String s = claims.get(JwtConstants.LOGIN_USER_ID, String.class);
            if (s == null) {
                return "";
            }
            return s;
        }
        return null;
    }

    /**
     * 裁剪token前缀
     */
    public static String replaceTokenPrefix(String token) {
        // 如果前端设置了令牌前缀，则裁剪掉前缀
        if (StrUtil.isNotEmpty(token) && token.startsWith(HttpConstants.PREFIX)) {
            token = token.replaceFirst(HttpConstants.PREFIX, StrUtil.EMPTY);
        }
        return token;
    }
}